Fixed ConvTranspose2d/3d breaking on newer PyTorch versions - #62
Fixed ConvTranspose2d/3d breaking on newer PyTorch versions#62InboraStudio wants to merge 3 commits into
Conversation
…t with torch>=2.9
|
Thanks for the fix @InboraStudio. Good catch and looks good to go. |
I have done that thanks for letting me know |
Thanks for the fix! The tests written assert that items in the Could you remove these tests and write tests in Also, the CI is failing with the following error: |
…e2d/3d axis resolution
You were right I misread the overload name .default doesn't exist for conv_transpose2d/3d in torch 2.8–2.11 |
|
I'm seeing that the only change in the PR is adding Where is |
Hey, noticed that conv_transpose2d and conv_transpose3d were using an older
.inputoverload in the ATEN op mapping that got deprecated in PyTorch 2.9.This meant any model with transpose convolution layers would silently break
during graph-mode quantization on newer PyTorch installs
Switched both over to the
.defaultoverload which is the standard now,and kept a small fallback in place so it still works fine on PyTorch 2.8.x
Should be a safe, non-breaking change across the full supported version range